home *** CD-ROM | disk | FTP | other *** search
/ CICA 1995 September (Japanese) / CICA Shareware for Windows CD-ROM (Walnut Creek) (September 1995) (Japanese) (Disc 2).iso / disc2 / nt / source.exe / POSIX / BSDPSX / SETENV.C < prev    next >
Encoding:
C/C++ Source or Header  |  1992-10-12  |  4.0 KB  |  123 lines
  1. /*
  2.  * Copyright (c) 1987 Regents of the University of California.
  3.  * All rights reserved.
  4.  *
  5.  * Redistribution and use in source and binary forms, with or without
  6.  * modification, are permitted provided that the following conditions
  7.  * are met:
  8.  * 1. Redistributions of source code must retain the above copyright
  9.  *    notice, this list of conditions and the following disclaimer.
  10.  * 2. Redistributions in binary form must reproduce the above copyright
  11.  *    notice, this list of conditions and the following disclaimer in the
  12.  *    documentation and/or other materials provided with the distribution.
  13.  * 3. All advertising materials mentioning features or use of this software
  14.  *    must display the following acknowledgement:
  15.  *    This product includes software developed by the University of
  16.  *    California, Berkeley and its contributors.
  17.  * 4. Neither the name of the University nor the names of its contributors
  18.  *    may be used to endorse or promote products derived from this software
  19.  *    without specific prior written permission.
  20.  *
  21.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31.  * SUCH DAMAGE.
  32.  */
  33.  
  34. #ifdef DF_POSIX
  35. #include <bsdlib.h>
  36. #endif
  37.  
  38. #if defined(LIBC_SCCS) && !defined(lint)
  39. static char sccsid[] = "@(#)setenv.c    5.6 (Berkeley) 6/4/91";
  40. #endif /* LIBC_SCCS and not lint */
  41.  
  42. #include <stddef.h>
  43. #include <stdlib.h>
  44. #include <string.h>
  45.  
  46. /*
  47.  * setenv --
  48.  *    Set the value of the environmental variable "name" to be
  49.  *    "value".  If rewrite is set, replace any current value.
  50.  */
  51. setenv(name, value, rewrite)
  52.     register const char *name;
  53.     register const char *value;
  54.     int rewrite;
  55. {
  56.     extern char **environ;
  57.     static int alloced;            /* if allocated space before */
  58.     register char *C;
  59.     int l_value, offset;
  60.     char *_findenv();
  61.  
  62.     if (*value == '=')            /* no `=' in value */
  63.         ++value;
  64.     l_value = strlen(value);
  65.     if ((C = _findenv(name, &offset))) {    /* find if already exists */
  66.         if (!rewrite)
  67.             return (0);
  68.         if (strlen(C) >= l_value) {    /* old larger; copy over */
  69.             while (*C++ = *value++);
  70.             return (0);
  71.         }
  72.     } else {                    /* create new slot */
  73.         register int    cnt;
  74.         register char    **P;
  75.  
  76.         for (P = environ, cnt = 0; *P; ++P, ++cnt);
  77.         if (alloced) {            /* just increase size */
  78.             environ = (char **)realloc((char *)environ,
  79.                 (size_t)(sizeof(char *) * (cnt + 2)));
  80.             if (!environ)
  81.                 return (-1);
  82.         }
  83.         else {                /* get new space */
  84.             alloced = 1;        /* copy old entries into it */
  85.             P = (char **)malloc((size_t)(sizeof(char *) *
  86.                 (cnt + 2)));
  87.             if (!P)
  88.                 return (-1);
  89.             bcopy(environ, P, cnt * sizeof(char *));
  90.             environ = P;
  91.         }
  92.         environ[cnt + 1] = NULL;
  93.         offset = cnt;
  94.     }
  95.     for (C = (char *)name; *C && *C != '='; ++C);    /* no `=' in name */
  96.     if (!(environ[offset] =            /* name + `=' + value */
  97.         malloc((size_t)((int)(C - name) + l_value + 2))))
  98.         return (-1);
  99.     for (C = environ[offset]; (*C = *name++) && *C != '='; ++C)
  100.         ;
  101.     for (*C++ = '='; *C++ = *value++; )
  102.         ;
  103.     return (0);
  104. }
  105.  
  106. /*
  107.  * unsetenv(name) --
  108.  *    Delete environmental variable "name".
  109.  */
  110. void
  111. unsetenv(name)
  112.     const char    *name;
  113. {
  114.     extern char **environ;
  115.     register char **P;
  116.     int offset;
  117.  
  118.     while (_findenv(name, &offset))        /* if set multiple times */
  119.         for (P = &environ[offset];; ++P)
  120.             if (!(*P = *(P + 1)))
  121.                 break;
  122. }
  123.